package com.atguigu.lease.common.utils;

/**
 * @author XC
 * @version 1.0
 * @description
 * @knowledge
 */

import com.atguigu.lease.common.exception.LeaseException;
import com.atguigu.lease.common.result.ResultCodeEnum;
import io.jsonwebtoken.*;
import io.jsonwebtoken.security.Keys;
import io.jsonwebtoken.security.SignatureException;

import javax.crypto.SecretKey;
import java.util.Date;

/**
 * token：包含三部分，头，负载，签名，其中传递的信息在负载中
 * 负载：在 token中负责保存登录用户的相关信息(userName账号，userId唯一标识)
 */
public class JwtUtil {
    private static final Long expire = 1000 * 60 * 60l;
    // 定义密钥签名，密码是 123456，已使用 hmac-256位加密，所以是 32位
    private static final SecretKey secretKey = Keys.hmacShaKeyFor("b8ad08a3a547e35829b821b75370301dd8c4b06bdd7771f9b541a75914068718".getBytes());

    // 1. 创建 token
    public static String createToken(String userName, Long userId) {
        String token = Jwts.builder()
                // 设置主题
                .setSubject("login-token")
                // 设置过期时间
                .setExpiration(new Date(System.currentTimeMillis() + expire))
                // 设置负载内容
                .claim("userName", userName)
                .claim("userId", userId)
                // 设置密钥签名
                .signWith(secretKey)
                // 压缩 token，减少传输大小
                .compressWith(CompressionCodecs.GZIP)
                // 用于生成 token
                .compact();

        return token;
    }

    // 2. 解析 token
    public static Claims parseToken(String token) {
        Jws<Claims> claimsJws = null;
        try {
            claimsJws = Jwts.parserBuilder()
                    // 设置签名，拿到签名中的密钥才能解析
                    .setSigningKey(secretKey)
                    .build()
                    // 解析 token
                    .parseClaimsJws(token);
        } catch (ExpiredJwtException e) {
            throw new LeaseException(ResultCodeEnum.TOKEN_EXPIRED);
        } catch (JwtException e) {
            throw new LeaseException(ResultCodeEnum.TOKEN_INVALID);
        }

        return claimsJws.getBody();
    }

    public static void main(String[] args) {
        // 测试 token
        String token = JwtUtil.createToken("admin", 1L);
        System.out.println(token);

        Claims claims = parseToken(token);
        System.out.println(claims);
    }
}
